python -m fit --dir $1 --save_folder $2 --cuda True